from fastapi import APIRouter, Depends, Security
from fastapi.responses import JSONResponse

from myProfile.models.data.myProfile_models import Login, Medication
from myProfile.repository.medications import MedicationRepository
from myProfile.models.request.myProfile_request import MedicationReq
from myProfile.db_config.odmantic_config import create_db_engine
from myProfile.security.secure import get_current_valid_user

router = APIRouter()


@router.post("/medication/add")
async def add_medication(req: MedicationReq,
                         current_user: Login = Security(
                             get_current_valid_user, scopes=["user"]),
                         engine=Depends(create_db_engine)):
    medication = Medication(**req.model_dump())
    repo: MedicationRepository = MedicationRepository(engine)
    result = await repo.add_to_profile(current_user.id, medication)
    if result is True:
        return req
    else:
        return JSONResponse(
            content={"message": "add medication problem encountered."},
            status_code=500)


@router.delete("/medication/delete/{id}")
async def delete_medication(id: int,
                            current_user: Login = Security(
                                get_current_valid_user, scopes=["user"]),
                            engine=Depends(create_db_engine)):
    repo: MedicationRepository = MedicationRepository(engine)
    result = await repo.remove_from_profile(current_user.id, id)
    if result is True:
        return JSONResponse(
            content={"message": "delete medication successfully"},
            status_code=201)
    else:
        return JSONResponse(
            content={"message": "delete medication unsuccessfully"},
            status_code=500)


@router.get("/medication/get")
async def get_medication(current_user: Login = Security(
                            get_current_valid_user, scopes=["user"]),
                         engine=Depends(create_db_engine)):
    repo: MedicationRepository = MedicationRepository(engine)
    return await repo.get_from_profile(current_user.id)
